package com.wanlian.util;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.apache.log4j.Logger;


/**
 * 获取MAC地址
 * 2017-04-10
 */
public class MacAddressUtil {
    protected static Logger logger = Logger.getLogger(MacAddressUtil.class);

    private static String callCmd(String[] cmd) {
        String result = "";
        String line = "";
        Process proc = null;

        try {
            proc = Runtime.getRuntime().exec(cmd);
            InputStreamReader is = new InputStreamReader(proc.getInputStream());
            BufferedReader br = new BufferedReader(is);

            while ((line = br.readLine()) != null) {
                result += line;
            }
        } catch (IOException e) {
            logger.error("获取Mac地址：执行命令行异常", e);
        }
        return result;
    }

    /**
     * @param cmd     第一个命令
     * @param another 第二个命令
     * @return 第二个命令的执行结果
     */
    private static String callCmd(String[] cmd, String[] another) {
        String result = "";
        String line = "";
        Runtime rt = Runtime.getRuntime();

        try {
            Process proc = rt.exec(cmd);
            proc.waitFor();                    // 等待，准备执行第二个命令
            proc = rt.exec(another);
            InputStreamReader is = new InputStreamReader(proc.getInputStream());
            BufferedReader br = new BufferedReader(is);
            while ((line = br.readLine()) != null) {
                result += line;
            }
        } catch (IOException | InterruptedException e) {
            logger.error("获取Mac地址：执行命令行异常", e);
        }
        return result;
    }


    /**
     * @param ip           目标ip,一般在局域网内
     * @param sourceString 命令处理的结果字符串
     * @param macSeparator mac分隔符号
     * @return mac地址，用上面的分隔符号表示
     */
    private static String filterMacAddress(final String ip, final String sourceString, final String macSeparator) {
        String result = "";
        String regExp = "((([0-9,A-F,a-f]{1,2}" + macSeparator + "){1,5})[0-9,A-F,a-f]{1,2})";
        Pattern pattern = Pattern.compile(regExp);
        Matcher matcher = pattern.matcher(sourceString);

        while (matcher.find()) {
            result = matcher.group(1);
            if (sourceString.indexOf(ip) <= sourceString.lastIndexOf(matcher.group(1))) {
                break; // 如果有多个IP,只匹配本IP对应的Mac.
            }
        }

        return result;
    }


    /**
     * @param ip 目标ip
     * @return Mac Address
     */
    private static String getMacInWindows(final String ip) {
        String result = "";
        String[] cmd = {
                "cmd",
                "/c",
                "ping " + ip
        };
        String[] another = {
                "cmd",
                "/c",
                "arp -a"
        };

        String cmdResult = callCmd(cmd, another);
        result = filterMacAddress(ip, cmdResult, "-");

        return result;
    }

    /**
     * @param ip 目标ip
     * @return Mac Address
     */
    private static String getMacInLinux(final String ip) {
        String result = "";
        String[] cmd = {
                "/bin/sh",
                "-c",
                "ping " + ip + " -c 2 && arp -a"
        };

        String cmdResult = callCmd(cmd);
        result = filterMacAddress(ip, cmdResult, ":");

        return result;
    }


    public static String tidyMac(String mac) {
        String macAddress = mac;
        if (!macAddress.contains(":")) {
            String regex = "(.{2})";
            macAddress = macAddress.replaceAll(regex, "$1:");
            macAddress = macAddress.substring(0, macAddress.length() - 1);
        }
        return macAddress;
    }

    public static void main(String[] args) {
        System.out.println(tidyMac("78f234567890"));
    }

}